Cognitive correction and stimulus control